#!/usr/bin/env python

# Copyright (c) 2005-2018 Alexander Belchenko
# All rights reserved.
#
# Redistribution and use in source and binary forms,
# with or without modification, are permitted provided
# that the following conditions are met:
#
# * Redistributions of source code must retain
#   the above copyright notice, this list of conditions
#   and the following disclaimer.
# * Redistributions in binary form must reproduce
#   the above copyright notice, this list of conditions
#   and the following disclaimer in the documentation
#   and/or other materials provided with the distribution.
# * Neither the name of the author nor the names
#   of its contributors may be used to endorse
#   or promote products derived from this software
#   without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
# BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
# AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
# OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
# OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
# EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

'''Intel HEX file format hex2bin convertor utility.'''

from array import array

def encode(fin, fout):
    import intelhex
    import sys
    import os

    if not os.path.isfile(fin):
        txt = "ERROR: File not found: %s" % fin  # that's required to get not-so-dumb result from 2to3 tool
        print(txt)
        return 1

    try:
        h = intelhex.IntelHex(fin)
    except intelhex.HexReaderError:
        e = sys.exc_info()[1]     # current exception
        txt = "ERROR: bad HEX file: %s" % str(e)
        print(txt)
        return 1
    bin = h.tobinarray()
    enc = array('B')
    for b in bin:
        enc.append(b ^ 0x86)

    try:
        with open(fout, 'wb') as fobj:
            fobj.write(enc)
    except IOError:
        e = sys.exc_info()[1]     # current exception
        txt = "ERROR: Could not write to file: %s: %s" % (fout, str(e))
        print(txt)
        return 1
    
    return 0

VERSION = '1.0.0'

def main():
    import getopt
    import os
    import sys

    usage = '''Hex86 convertor utility.
Usage:
    python hex86.py [options] INFILE [OUTFILE]

Arguments:
    INFILE      name of hex file for processing.
    OUTFILE     name of output file. If omitted then output
                will be writing to stdout.

Options:
    -h, --help              this help message.
    -v, --version           version info.
'''

    try:
        opts, args = getopt.getopt(sys.argv[1:], "hv",
                                  ["help", "version"])

        for o, a in opts:
            if o in ("-h", "--help"):
                print(usage)
                sys.exit(0)
            elif o in ("-v", "--version"):
                print(VERSION)
                sys.exit(0)
 
        if not args:
            raise getopt.GetoptError('Hex file is not specified')

        if len(args) > 2:
            raise getopt.GetoptError('Too many arguments')

    except getopt.GetoptError:
        msg = sys.exc_info()[1]     # current exception
        txt = 'ERROR: '+str(msg)  # that's required to get not-so-dumb result from 2to3 tool
        print(txt)
        print(usage)
        sys.exit(2)

    fin = args[0]
    if not os.path.isfile(fin):
        txt = "ERROR: File not found: %s" % fin  # that's required to get not-so-dumb result from 2to3 tool
        print(txt)
        sys.exit(1)

    fout = args[1]
    
    return encode(fin, fout)

# def main():
#     import getopt
#     import os
#     import sys

#     usage = '''Hex86 convertor utility.
# Usage:
#     python hex86.py [options] INFILE [OUTFILE]

# Arguments:
#     INFILE      name of hex file for processing.
#     OUTFILE     name of output file. If omitted then output
#                 will be writing to stdout.

# Options:
#     -h, --help              this help message.
#     -v, --version           version info.
# '''

#     try:
#         opts, args = getopt.getopt(sys.argv[1:], "hv",
#                                   ["help", "version"])

#         for o, a in opts:
#             if o in ("-h", "--help"):
#                 print(usage)
#                 sys.exit(0)
#             elif o in ("-v", "--version"):
#                 print(VERSION)
#                 sys.exit(0)
 
#         if not args:
#             raise getopt.GetoptError('Hex file is not specified')

#         if len(args) > 2:
#             raise getopt.GetoptError('Too many arguments')

#     except getopt.GetoptError:
#         msg = sys.exc_info()[1]     # current exception
#         txt = 'ERROR: '+str(msg)  # that's required to get not-so-dumb result from 2to3 tool
#         print(txt)
#         print(usage)
#         sys.exit(2)

#     fin = args[0]
#     if not os.path.isfile(fin):
#         txt = "ERROR: File not found: %s" % fin  # that's required to get not-so-dumb result from 2to3 tool
#         print(txt)
#         sys.exit(1)

#     fout = args[1]

#     import intelhex
    
#     try:
#         h = intelhex.IntelHex(fin)
#     except intelhex.HexReaderError:
#         e = sys.exc_info()[1]     # current exception
#         txt = "ERROR: bad HEX file: %s" % str(e)
#         print(txt)
#         return 1
#     bin = h.tobinarray()
#     enc = array('B')
#     for b in bin:
#         enc.append(b ^ 0x86)

#     try:
#         with open(fout, 'wb') as fobj:
#             fobj.write(enc)
#     except IOError:
#         e = sys.exc_info()[1]     # current exception
#         txt = "ERROR: Could not write to file: %s: %s" % (fout, str(e))
#         print(txt)
#         return 1

#     return 0



if __name__ == '__main__':
    main()
